import torch


class Task:
    def __init__(self, id):
        self.compute_size = 0
        self.id = id
        self.deadline = 0

    def length(self):
        return self.compute_size

    def get_feature(self) -> torch.Tensor:
        return [self.compute_size, self.deadline]


# if __name__ == '__main__':
#     t1 = Task()
#     t2 = Task()
#     t2.deadline = 3
#     f1 = t1.get_feature()
#     f2 = t2.get_feature()
#     print(torch.vstack([f1, f2]))
